#!/usr/bin/env python

from le_mensagens import *
from dicionario import Dicionario

n1 = 500
n2 = 200
n3 = 100

mensagens = leMensagens() # Le as mensagens e cria uma lista de listas a partir delas.

d = Dicionario(mensagens) # Cria um dicionario contendo todas as palavras (filtradas) presentes nas mensagens.
dicio500 = d.selecionaPalavras(n1)

print('Gerando vetores descritores...')
arq500 = open('../saida/desc500','w')
arq200 = open('../saida/desc200','w')
arq100 = open('../saida/desc100','w')
for mensagem in mensagens:
   desc500 = [0] * n1
   for palavra in mensagem:
      try:
         indice=dicio500.index(palavra)
         desc500[indice]+=1
      except ValueError:
         continue
   desc200 = desc500[-n2:]
   desc100 = desc500[-n3:]
   desc500 = (' '.join(map(str,desc500)))
   desc200 = (' '.join(map(str,desc200)))
   desc100 = (' '.join(map(str,desc100)))
   #print(desc500)
   #print(desc200)
   #print(desc100)
   arq500.write(desc500 + '\n')
   arq200.write(desc200 + '\n')
   arq100.write(desc100 + '\n')
arq500.close()
arq200.close()
arq100.close()
print('Vetores descritores gerados. Script encerrado.')
